import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# 数据库配置
DATABASE_CONFIG = {
    'host': os.getenv('DB_HOST', 'localhost'),
    'port': os.getenv('DB_PORT', '5432'),
    'database': os.getenv('DB_NAME', 'yh'),
    'username': os.getenv('DB_USER', 'postgres'),
    'password': os.getenv('DB_PASSWORD', '123456')
}

# LLM配置
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY', '')
OPENAI_BASE_URL = os.getenv('OPENAI_BASE_URL', 'https://api.openai.com/v1')

# 新增：模型参数配置
LLM_CONFIG = {
    'base_url': 'https://ai.nengyongai.cn/v1',
    'openai_api_key': 'sk-1UFn4uObTYmMq2J38sSnSgwQT6sQOmCcIVmkL8fUCR8X0BLA',
    'model_name': 'gpt-4o-mini',
    'temperature': 0,
    'max_tokens': None
}

# 地图配置
MAP_CONFIG = {
    'default_center': [39.9042, 116.4074],  # 北京
    'default_zoom': 10,
    'tile_server': 'OpenStreetMap',
    'output_dir': 'outputs',
    'temp_dir': 'temp'
}

# 创建输出目录
os.makedirs(MAP_CONFIG['output_dir'], exist_ok=True)
os.makedirs(MAP_CONFIG['temp_dir'], exist_ok=True) 